# 定义对象库的基类
from selenium.webdriver.support.wait import WebDriverWait

from utils import UtilsDriver


class BasePage:
    def __init__(self):
        self.driver = UtilsDriver.get_driver()  # 获取驱动对象

    # 定义获取元素对象的方法
    def get_element(self, location):
        """
        :param location: 元素的定位信息
        :return:  返回元素对象
        """
        wait = WebDriverWait(self.driver, 10)
        return wait.until(lambda x: x.find_element(*location))


# 定义操作层基类
class BaseHandle:
    # 定义元素的输入操作方法
    def input_text(self, element, text):
        """
        :param element:  代表的是元素对象
        :param text:  代表的是输入的文本信息
        :return:
        """
        element.clear()  # 清除元素对象中的内容
        element.send_keys(text)  # 输入文本内容。
